# 模型
from .exts import  db

# 模型
class User(db.Model):
    # 表名
    __tablename__ = "tb_user"
    id = db.Column(db.Integer, primary_key=True,autoincrement=True)
    username = db.Column(db.String(80), unique=True, nullable=False)
    password = db.Column(db.String(128), nullable=False)


class Config(db.Model):
    # 表名
    __tablename__ = "tb_sysconfig"
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)  # 主键
    name = db.Column(db.String(50), nullable=False)
    key = db.Column(db.String(50), unique=True, nullable=False)  # 配置键
    value = db.Column(db.String(255), nullable=False)  # 配置值
    description = db.Column(db.String(255))  # 描述

class HistoryDetection(db.Model):
    # 表名
    __tablename__ = "tb_history"
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)  # 主键
    results = db.Column(db.Text, unique=True, nullable=False)  # 结果
    processed_image = db.Column(db.String(255), unique=True, nullable=False)  #模型保存结果
    image = db.Column(db.String(255), unique=True, nullable=False)  #原始图片
    create_time = db.Column(db.String(50), unique=True, nullable=False)  #时间

    # 将模型对象转换为字典
    def to_dict(self):
        return {
            'id': self.id,
            'results': self.results,
            'processed_image': self.processed_image,
            'image': self.image,
            'create_time': self.create_time
        }

